﻿// least_recently_used.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

#include <iostream>
#include <map>

using namespace std;

class LRUCache {
public:
    LRUCache(int capacity) {
        head = new Node(int(), int());
        tail = new Node(int(), int());

        head->next = tail;
        tail->prev = head;

        capacity_ = capacity;
        listsize_ = 0;

    }

    int get(int key) {
        if (my_map.count(key))
        {
            Node* curr_node = my_map[key];
            moveToFirst(curr_node);
            
            return curr_node->map_value;
        }
        return -1;
    }

    void put(int key, int value) {
        if (my_map.count(key))
        {
            Node* curr_node = my_map[key];
            curr_node->map_value = value;
            moveToFirst(curr_node);
        }
        else
        {
            addFirst(key, value);
            listsize_++;
            if (listsize_ > capacity_)
            {
                removeLast();
                listsize_--;
            }
        }
    }
private:
    struct Node {
        int map_key;
        int map_value;
        Node* next;
        Node* prev;
        Node(int key, int value) :map_key(key), map_value(value), next(nullptr), prev(nullptr) {};
    };
    Node* head;
    Node* tail;
    int listsize_;
    int capacity_;
    map<int, Node*>my_map;

    void addFirst(int key, int value){
        
        Node* next_node = head->next;
        Node* new_node = new Node(key, value);

        head->next = new_node;
        new_node->prev = head;

        if (next_node)
        {
            next_node->prev = new_node;
            new_node->next = next_node;
        }
        else {
            new_node->next = nullptr;
        }
        //注意，插入不要使用insert，insert不会覆盖已存在的键值
        my_map[key] = new_node;
        
    }

    void moveToFirst(Node* curr_node){
        // 前置判断 1my_map是否存在 2是否修改my_map中的value
        // 这里只负责删节点+移到开头
        
        Node* prev_node = curr_node->prev;
        Node* next_node = curr_node->next;

        int move_key = curr_node->map_key;
        int move_value= curr_node->map_value;
        
        if (next_node)
        {
            prev_node->next = next_node;
            next_node->prev = prev_node;
        }
        else {
            prev_node->next = nullptr;
        }

        addFirst(move_key, move_value);
    }


    void removeLast(){
        Node* del_node = tail->prev;
        Node* prev_node = del_node->prev;

        int del_key = del_node->map_key;
        my_map.erase(del_key);

        prev_node->next = tail;
        tail->prev = prev_node;
    }

};

int main()
{
    LRUCache list(2);
    list.put(1, 1);
    list.put(2, 2);
    list.get(1);
    list.put(3, 3);
    list.get(2);

}

// 运行程序: Ctrl + F5 或调试 >“开始执行(不调试)”菜单
// 调试程序: F5 或调试 >“开始调试”菜单

// 入门使用技巧: 
//   1. 使用解决方案资源管理器窗口添加/管理文件
//   2. 使用团队资源管理器窗口连接到源代码管理
//   3. 使用输出窗口查看生成输出和其他消息
//   4. 使用错误列表窗口查看错误
//   5. 转到“项目”>“添加新项”以创建新的代码文件，或转到“项目”>“添加现有项”以将现有代码文件添加到项目
//   6. 将来，若要再次打开此项目，请转到“文件”>“打开”>“项目”并选择 .sln 文件
